package com.john.sqwen.ecbc.core.config;

import com.john.sqwen.ecbc.commons.utils.StringUtils;
import org.apache.ibatis.io.VFS;
import org.apache.ibatis.session.SqlSessionFactory;
import org.mybatis.spring.SqlSessionFactoryBean;
import org.mybatis.spring.boot.autoconfigure.SpringBootVFS;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.env.Environment;
import org.springframework.core.io.DefaultResourceLoader;
import org.springframework.core.io.Resource;
import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
import org.springframework.core.io.support.ResourcePatternResolver;
import org.springframework.core.type.classreading.CachingMetadataReaderFactory;
import org.springframework.core.type.classreading.MetadataReader;
import org.springframework.core.type.classreading.MetadataReaderFactory;
import org.springframework.util.ClassUtils;

import javax.sql.DataSource;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashSet;
import java.util.List;

/**
 * @author john_sqwen
 * @packageName com.john.sqwen.ecbc.core.config
 * @date 2022/4/24
 * @projectName ecbc-base
 * @Description MyBatisConfig
 **/
@Configuration
public class MyBatisConfig {

        @Autowired
        private Environment env;

        static final String DEFAULT_RESOURCE_PATTERN = "**/*.class";

        public static String setTypeAliasesPackage(String typeAliasesPackage)
        {
            ResourcePatternResolver resolver = (ResourcePatternResolver) new PathMatchingResourcePatternResolver();
            MetadataReaderFactory metadataReaderFactory = new CachingMetadataReaderFactory(resolver);
            List<String> allResult = new ArrayList<String>();
            try
            {
                for (String aliasesPackage : typeAliasesPackage.split(","))
                {
                    List<String> result = new ArrayList<String>();
                    aliasesPackage = ResourcePatternResolver.CLASSPATH_ALL_URL_PREFIX
                            + ClassUtils.convertClassNameToResourcePath(aliasesPackage.trim()) + "/" + DEFAULT_RESOURCE_PATTERN;
                    Resource[] resources = resolver.getResources(aliasesPackage);
                    if (resources != null && resources.length > 0)
                    {
                        MetadataReader metadataReader = null;
                        for (Resource resource : resources)
                        {
                            if (resource.isReadable())
                            {
                                metadataReader = metadataReaderFactory.getMetadataReader(resource);
                                try
                                {
                                    result.add(Class.forName(metadataReader.getClassMetadata().getClassName()).getPackage().getName());
                                }
                                catch (ClassNotFoundException e)
                                {
                                    e.printStackTrace();
                                }
                            }
                        }
                    }
                    if (result.size() > 0)
                    {
                        HashSet<String> hashResult = new HashSet<String>(result);
                        allResult.addAll(hashResult);
                    }
                }
                if (allResult.size() > 0)
                {
                    typeAliasesPackage = String.join(",", (String[]) allResult.toArray(new String[0]));
                }
                else
                {
                    throw new RuntimeException("mybatis typeAliasesPackage 路径扫描错误,参数typeAliasesPackage:" + typeAliasesPackage + "未找到任何包");
                }
            }
            catch (IOException e)
            {
                e.printStackTrace();
            }
            return typeAliasesPackage;
        }

        public Resource[] resolveMapperLocations(String[] mapperLocations)
        {
            ResourcePatternResolver resourceResolver = new PathMatchingResourcePatternResolver();
            List<Resource> resources = new ArrayList<Resource>();
            if (mapperLocations != null)
            {
                for (String mapperLocation : mapperLocations)
                {
                    try
                    {
                        Resource[] mappers = resourceResolver.getResources(mapperLocation);
                        resources.addAll(Arrays.asList(mappers));
                    }
                    catch (IOException e)
                    {
                        // ignore
                    }
                }
            }
            return resources.toArray(new Resource[resources.size()]);
        }

        @Bean
        public SqlSessionFactory sqlSessionFactory(DataSource dataSource) throws Exception
        {
            String typeAliasesPackage = env.getProperty("mybatis.typeAliasesPackage");
            String mapperLocations = env.getProperty("mybatis.mapperLocations");
            String configLocation = env.getProperty("mybatis.configLocation");
            typeAliasesPackage = setTypeAliasesPackage(typeAliasesPackage);
            VFS.addImplClass(SpringBootVFS.class);

            final SqlSessionFactoryBean sessionFactory = new SqlSessionFactoryBean();
            sessionFactory.setDataSource(dataSource);
            sessionFactory.setTypeAliasesPackage(typeAliasesPackage);
            sessionFactory.setMapperLocations(resolveMapperLocations(StringUtils.split(mapperLocations, ",")));
            sessionFactory.setConfigLocation(new DefaultResourceLoader().getResource(configLocation));
            return sessionFactory.getObject();
        }
}
